package 栈算法;

import java.util.Deque;
import java.util.LinkedList;
import java.util.List;

public class postorderTraversal {
    /**
     * 二叉树的后序遍历
     * 先沿着最左边这条路走到最左节点, 并将访问过的节点加入stack
     * 如图, 访问完最左节点后, 他的父节点出栈, 但是在访问之前要先检查父节点的右孩子是否已经访问, 如果没有的话, 父节点入栈搁着, 先访问他的右孩子
     *
     * @param root
     * @return
     */
    public List<Integer> postorderTraversal(TreeNode root) {
        List<Integer> res = new LinkedList<>();
        if (root == null) {
            return res;
        }

        TreeNode preVisited = null; // 刚刚已经访问过的节点
        Deque<TreeNode> stk = new LinkedList<>();

        while (!stk.isEmpty() || root != null) {
            while (root != null) {
                stk.push(root);
                root = root.left;
            }
            root = stk.pop();
            //如果当前节点的右孩子不为空 并且 当前节点的右孩子之前未被访问
            if (root.right != null && root.right != preVisited) {
                stk.push(root);
                root = root.right;
            } else {
                // 左右根
                res.add(root.val);
                preVisited = root;
                root = null; // 防止进入内while循环
            }
        }
        return res;
    }

    public class TreeNode {
        int val;
        TreeNode left;
        TreeNode right;

        TreeNode() {
        }

        TreeNode(int val) {
            this.val = val;
        }

        TreeNode(int val, TreeNode left, TreeNode right) {
            this.val = val;
            this.left = left;
            this.right = right;
        }
    }
}
